package com.sxj.shop.util;

import com.alibaba.druid.pool.DruidDataSourceFactory;
import com.sxj.shop.exception.DataAccessException;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Properties;

/**
 * 线程单例的连接池
 */
public class JDBCUtil {

    // 数据库连接池
    private static DataSource ds;
    // 线程池,用于实现线程单例
    private static ThreadLocal<Connection> local;

    static {
        local = new ThreadLocal<>();
        Properties p = new Properties();
        try {
            p.load(JDBCUtil.class.getClassLoader().getResourceAsStream("dataSource.properties"));
            // druid
            ds = DruidDataSourceFactory.createDataSource(p);
        } catch (Exception e) {
            e.printStackTrace();
            throw new ExceptionInInitializerError("JDBCUtil初始化失败");
        }
    }

    /**
     * 获取连接
     * 使用数据库连接池获取一个线程单例的连接
     * @return
     */
    public static Connection getConnection() {
        // 1.从线程池中取出连接
        Connection conn = local.get();

        try {
            // 2.如果线程池中没有连接
            if (conn == null) {
                // 3.则从数据库连接池中取出一个连接
                conn = ds.getConnection();
                // 4.当拿到连接之后,将该连接存放到线程池,便于实现线程单例
                // 如果下次再使用链接,则直接从线程池中获取连接
                local.set(conn);
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("获取连接失败");
        }
        // 2.如果线程池中获取到了连接,则说明一定是之前存放的,直接返回,保证是同一个
        // 5.如果线程池中没有获取连接到连接,从数据库连接池中取出了连接返回出去
        return conn;
    }

    /**
     * 释放持久层资源,根据实际情况进行释放,如果某些资源不需要释放,则赋值为null即可
     * @param conn
     * @param ps
     * @param rs
     */
    public static void close(Connection conn, PreparedStatement ps, ResultSet rs) {
        if (conn != null) {
            try {
                conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("释放连接失败");
            }
        }
        if (ps != null) {
            try {
                ps.close();
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("释放状态集失败");
            }
        }
        if (rs != null) {
            try {
                rs.close();
            } catch (SQLException e) {
                e.printStackTrace();
                throw new DataAccessException("释放结果集失败");
            }
        }
    }

    /**
     * 释放线程池中的连接
     */
    public static void close() {
        // 1.从线程池中取出需要释放的连接
        Connection conn = local.get();
        try {
            if (conn != null) {
                // 2.释放连接
                conn.close();
                // 3.对于一个业务整体而言,当业务结束之后,就不需要连接了
                // 因此可以移除线程池中的连接
                local.remove();
            }
        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("释放线程池资源失败");
        }
        // 如果没有从线程池中取到连接,则不需要做任何处理
    }

}
